#include<iostream>
#include<cstdlib>
#include<unistd.h>
#include<strings.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>

using namespace std;

void Usage(string proc)
{
    cout<<"\n\rUsage: "<<proc<<"serverip serverport\n"<<endl;
}

// ./udpclient serverip serverport总共接受三个参数
int main(int argc,char*argv[])
{
    if(argc!=3)
    {
        Usage(argv[0]);
        exit(0);
    }
    string serverip=argv[1];
    uint16_t serverport=stoi(argv[2]);
    struct sockaddr_in server;
    bzero(&server,sizeof(server));
    server.sin_family=AF_INET;
    server.sin_port=htons(serverport);//将主机字节序转化为网络字节序
    server.sin_addr.s_addr=inet_addr(serverip.c_str());//将点分十进制的 IP 字符串转换为网络字节序的整数
    socklen_t len=sizeof(server);
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(sockfd<0)
    {
        cout<<"socker error"<<endl;
        return 1;
    }

    string message;
    char buffer[1024];
    while(true)
    {
        cout<<"please enter@ ";
        getline(cin,message);//从标准输入读取用户输入的消息
        //将消息发送到服务器的地址（server结构体）
        sendto(sockfd,message.c_str(),message.size(),0,(struct sockaddr*)&server,len);

        struct sockaddr_in temp;
        socklen_t len=sizeof(temp);
        //接收服务器返回的响应数据
        ssize_t s=recvfrom(sockfd,buffer,1023,0,(struct sockaddr*)&temp,&len);
        if(s>0)
        {
            buffer[s]=0;
            cout<<buffer<<endl;
        }
    }
    close(sockfd);
    return 0;
}